/*
 * Copyright (C) 2010 Fores Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package jp.fores.midori.server;

import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;

/**
 * gzip用の出力ストリームを返すようにしたHttpServletResponseクラス。<br>
 */
public class GZipHttpServletResponse extends HttpServletResponseWrapper {
	//==========================================================
	//フィールド

	/**
	 * サーブレット出力ストリーム
	 */
	protected ServletOutputStream os = null;

	/**
	 * プリントライター
	 */
	protected PrintWriter writer = null;


	//==========================================================
	//メソッド

	/**
	 * コンストラクタです。<br>
	 *
	 * @param response レスポンス
	 */
	public GZipHttpServletResponse(HttpServletResponse response) {
		//親クラスのコンストラクタをそのまま呼び出す
		super(response);
	}


	//_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/
	//オーバーライドするメソッド

	/**
	 * {@inheritDoc}
	 */
	@Override
	public ServletOutputStream getOutputStream() throws IOException {
		//まだフィールドのサーブレット出力ストリームが設定されていない場合
		if (this.os == null) {
			//親クラスが返すサーブレット出力ストリームをGZipServletOutputStreamでラッピングして、フィールドに設定する
			this.os = new GZipServletOutputStream(super.getOutputStream());
		}

		//フィールドのサーブレット出力ストリームを返す
		return this.os;
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public PrintWriter getWriter() throws IOException {
		//まだフィールドのプリントライターが設定されていない場合
		if (this.writer == null) {
			//出力ストリームをプリントライターに変換してフィールドに設定する
			this.writer = new PrintWriter(new OutputStreamWriter(getOutputStream(), getCharacterEncoding()));
		}

		//フィールドのプリントライターを返す
		return this.writer;
	}
}
